package ERMDE.styling.tests;


import main.java.ERMDE.models.EntityType;
import main.java.ERMDE.styling.StyleConstants;
import main.java.ERMDE.styling.Styler;

import org.junit.Assert;
import org.junit.Test;


public class StylerTestCase {

	@Test
	public void testShouldReturnCorrectColorBasedOnEntityType()
			throws Exception {
		Assert.assertEquals(StyleConstants.DOMAIN_COLOR,
				Styler.getFillColor(EntityType.Domain));
		Assert.assertEquals(StyleConstants.THING_COLOR,
				Styler.getFillColor(EntityType.Thing));
		Assert.assertEquals(StyleConstants.HISTORIC_COLOR,
				Styler.getFillColor(EntityType.Historic));
		Assert.assertEquals(StyleConstants.PROGRAMMED_COLOR,
				Styler.getFillColor(EntityType.Programmed));
		Assert.assertEquals(StyleConstants.NONE_COLOR,
				Styler.getFillColor(EntityType.None));
	}
}
